cd ..
python -u train.py \
          --model_name unet_aws\
          --train_file /home/zhengyu/dataset/PWAFSDataset/PWAFS_train.csv\
          --val_file /home/zhengyu/dataset/PWAFSDataset/PWAFS_val.csv\
          --test_file /home/zhengyu/dataset/PWAFSDataset/PWAFS_test.csv\
          --root_dir /home/zhengyu/dataset/PWAFSDataset\
          --gpus 0,1,2,3\
          --num_works 32\
          --width 120\
          --batch_size 2\
          --max_epochs 500\
          --learning_rate 0.001\
          --downscale_factor 3\
          --input_length 19\
          --target_length 13\
          --input_channels 24\
          --predict_channels 1\
          --tensorboard_save_path ./lightning_logs/\
          --tensorboard_exp_name unet_aws\
          --loss_fx weighted_l1\
          --save_monitor epoch_val_loss\
          --save_dirpath ./save/unet_aws\
          --save_filename Unet_aws_{epoch:03d}-{epoch_val_loss:.3f}\
          --save_mode min\
          --save_top_k 500\
          --val_log_steps 20\
          --weights_prec 1 1 2.5 5 10 20\
          --thresholds_prec 0.0083 0.0167 0.083 0.167 0.333\
          --check_val_rate 1\
